-
Notifications
You must be signed in to change notification settings - Fork 74k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Porting the Conv kernel from Tensorflow Lite to Tensorflow Lite Micro #27362
Porting the Conv kernel from Tensorflow Lite to Tensorflow Lite Micro #27362
Conversation
I believe we have some engineers already working on this kernel. @petewarden can offer guidance. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This is looking great! Thanks for your PR.
Please also add the corresponding entry in the BUILD file.
Here is my first round of code review, mainly focusing on the kernel implementation.
4784cc0
to
c2bb280
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Sry for the delay. This looks great! Thanks!
Would love to sync with you on the next plan to avoid duplication of efforts. What are the operators missing in your use case that you would like the support next? Thanks
@@ -0,0 +1,329 @@ | |||
/* Copyright 2017 The TensorFlow Authors. All Rights Reserved. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
2019
Nits: next time, better not to use force push so that the comment history and diff can be preserved. |
@jenselo can you please address Ubuntu Sanity build failures. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Review done by @wangtz, just formally approving.
c2bb280
to
f9153a8
Compare
Yes, I'm looking into it today. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
@jenselo could you please fix build errors. Thanks ! |
PiperOrigin-RevId: 247806095
This PR ports the Conv kernel in Tensorflow Lite to Tensorflow Lite Micro.